package com.zhz.leetcode;

import com.zhz.leetcode.entity.TreeNode;

import java.util.Arrays;

/**
 * 108. 将有序数组转换为二叉搜索树
 * <p>
 * 给你一个整数数组 nums ，其中元素已经按 升序 排列，请你将其转换为一棵 高度平衡 二叉搜索树。
 * 高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
 * <p>
 * 示例 1：
 * 输入：nums = [-10,-3,0,5,9]
 * 输出：[0,-3,9,-10,null,5]
 * 解释：[0,-10,5,null,-3,null,9] 也将被视为正确答案：
 * <p>
 * 示例 2：
 * 输入：nums = [1,3]
 * 输出：[3,1]
 * 解释：[1,3] 和 [3,1] 都是高度平衡二叉搜索树。
 * <p>
 * 提示：
 * 1 <= nums.length <= 104
 * -104 <= nums[i] <= 104
 * nums 按 严格递增 顺序排列
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class ConvertSortedArrayToBinarySearchTree108 {
    public static void main(String[] args) {
        ConvertSortedArrayToBinarySearchTree108 m = new ConvertSortedArrayToBinarySearchTree108();
        int[] nums = {-10, -3, 0, 5, 9};
        TreeNode node = m.sortedArrayToBST1(nums);
    }

    /**
     * 思路：先找到中间节点，然后往两边扩散放置
     * 同时创建新的数组保存两边的数据
     */
    public TreeNode sortedArrayToBST(int[] nums) {
        int rootIndex = nums.length / 2;
        TreeNode root = new TreeNode(nums[rootIndex]);
        int[] leftArr = new int[rootIndex];
        int[] rightArr = new int[nums.length - 1 - rootIndex];
        for (int i = 0; i < nums.length; i++) {
            if (i < rootIndex) {
                leftArr[i] = nums[i];
            } else if (i > rootIndex) {
                rightArr[i - rootIndex - 1] = nums[i];
            }
        }
        if (leftArr.length != 0) {
            root.left = sortedArrayToBST(leftArr);
        }
        if (rightArr.length != 0) {
            root.right = sortedArrayToBST(rightArr);
        }
        return root;
    }


    /**
     * 思路同上面的一样，但是不创建新数组保存数据
     */
    public TreeNode sortedArrayToBST1(int[] nums) {
        return test(nums, 0, nums.length - 1);
    }

    public TreeNode test(int[] nums, int left, int right) {
        if (left > right) {
            return null;
        }
        int rootIndex = (left + right) / 2;
        TreeNode root = new TreeNode(nums[rootIndex]);
        root.left = test(nums, left, rootIndex - 1);
        root.right = test(nums, rootIndex + 1, right);
        return root;
    }
}
